package com.ruoyi.redtea.dualprevent.controller;


import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.ruoyi.common.core.domain.PageQuery;
import com.ruoyi.common.core.web.controller.BaseController;
import com.ruoyi.common.core.web.domain.AjaxResult;
import com.ruoyi.redtea.dualprevent.domain.entity.TrainRegion;
import com.ruoyi.redtea.dualprevent.domain.vo.TrainRegionVo;
import com.ruoyi.redtea.dualprevent.service.TrainRegionService;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import java.util.List;

/**
 * 区域表(TrainRegion)表控制层
 *
 * @author makejava
 * @since 2024-05-13 14:19:27
 */
@RestController
@RequestMapping("trainRegion")
public class TrainRegionController extends BaseController {
    /**
     * 服务对象
     */
    @Resource
    private TrainRegionService trainRegionService;

    @Resource
    private TrainRegionService trainAssessAnalyseUnitService;

    /**
     * 分页查询所有数据
     *
     * @return 所有数据
     */
/*    @GetMapping
    public AjaxResult selectAll(Page<TrainRegion> page, TrainRegion trainRegion) {
        return success(this.trainRegionService.page(page, new QueryWrapper<>(trainRegion)));
    }*/
    @GetMapping("/list")
    public AjaxResult selectAll() {
        return success(this.trainRegionService.queryList());
    }
    @GetMapping("/listPage")
    public AjaxResult selectAll(PageQuery pageQuery, TrainRegion trainRegion) {
        List<TrainRegionVo> trainRegionVos = this.trainRegionService.queryPageList(pageQuery);
        return success(trainRegionVos);
    }

    /**
     * 通过主键查询单条数据
     *
     * @param id 主键
     * @return 单条数据
     */
    @GetMapping("{id}")
    public AjaxResult selectOne(@PathVariable Integer id) {
        return success(this.trainRegionService.queryById(id));
    }

    /**
     * 新增数据
     *
     * @param trainRegion 实体对象
     * @return 新增结果
     */
    @PostMapping
    public AjaxResult insert(@RequestBody TrainRegion trainRegion) {
        return success(this.trainRegionService.insert(trainRegion));
    }

    /**
     * 修改数据
     *
     * @param trainRegion 实体对象
     * @return 修改结果
     */
    @PutMapping
    public AjaxResult update(@RequestBody TrainRegion trainRegion) {
        boolean update = trainRegionService.update(trainRegion);
        if(update){
            return success(update);
        }else {
            return error("更新失败LGP");
        }
    }

    /**
     * 删除数据
     *
     * @param idList 主键结合
     * @return 删除结果
     */
    @DeleteMapping
    public AjaxResult delete(@RequestParam("idList") List<Long> idList) {
        return success(this.trainRegionService.deleteByIds(idList));
    }

    /**
     * 获取风险四色图
     * @return
     */
    @GetMapping("/riskFourColor/list")
    public AjaxResult getFColorList(){
        return success(trainRegionService.queryFour());
    }

}

